#pragma once

#include <iostream>
#include <string>
#include <ctime>
#include <atomic>
#include <fstream>

#include <sys/time.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    class time_util
    {
    public:
        // 获取具体的事件
        static std::string get_time()
        {
            // 基于当前系统的当前日期和事件
            time_t now = time(0);

            // 将now转换成为字符串形式
            char *dt = ctime(&now);

            std::string data(dt);

            // 因为ctime返回的字符串末尾有一个换行符,所以我们需要把他给删除掉
            data.erase(data.length() - 1);

            return data;
        }

        // 这里获得时间戳,单位是秒
        static std::string get_timestamp()
        {
            struct timeval time;
            gettimeofday(&time, nullptr);
            return std::to_string(time.tv_sec);
        }

        // 这里获取时间戳,单位是毫秒
        static std::string get_time_ms()
        {
            struct timeval time;
            gettimeofday(&time, nullptr);
            // 前面是把秒换算成毫秒,后面是把微秒换算成毫秒
            return std::to_string(time.tv_sec * 1000 + time.tv_usec / 1000);
        }
    };

    // temp文件夹的路径,是文件添加的前缀
    static const std::string temp_path = "./temp/";

    class path_util
    {
    public:
        static std::string add_suffix(const std::string &file_name,
                                      const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }

        // 编译时需要有的临时文件
        // 构建源文件路径+后缀的完成文件名
        // 1234 -> ./temp/1234.cpp
        static std::string src(const std::string &file_name)
        {
            return add_suffix(file_name, ".cpp");
        }

        // 构建可执行程序的完整路径+后缀名
        static std::string exe(const std::string &file_name)
        {
            return add_suffix(file_name, ".exe");
        }

        // 构建该程序对应的标准错误信息的完整路径+后缀名
        static std::string compiler_error(const std::string &file_name)
        {
            return add_suffix(file_name, ".compile_error");
        }

        // 运行时需要有的临时文件
        static std::string stdin(const std::string &file_name)
        {
            return add_suffix(file_name, ".stdin");
        }

        static std::string stdout(const std::string &file_name)
        {
            return add_suffix(file_name, ".stdout");
        }

        static std::string stderr(const std::string &file_name)
        {
            return add_suffix(file_name, ".stderr");
        }
    };

    class file_util
    {
    public:
        static bool if_file_exists(const std::string &file_name)
        {
            struct stat st;
            if (stat(file_name.c_str(), &st) == 0)
            {
                // 获取文件属性成功,说明文件是存在的
                return true;
            }
            return false;
        }

        static std::string unique_file_name()
        {
            // PS:这个类的拷贝构造是删除了的
            static std::atomic<int> file_id(0);
            // 原子操作
            ++file_id;

            // 毫秒时间戳+原子性递增唯一值:来保证文件名的唯一性
            std::string ms = time_util::get_time_ms();
            std::string unique_id = std::to_string(file_id);
            return ms + "_" + unique_id;
        }

        static bool write_file(const std::string &target, const std::string &content)
        {
            std::ofstream out(target);
            if (!out.is_open())
            {
                // 文件打开失败
                return false;
            }

            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }

        static bool read_file(const std::string &target, std::string *content, bool retain_LF = false)
        {
            std::ifstream in(target);
            if (!in.is_open())
            {
                // 文件打开失败
                return false;
            }

            std::string line;
            // getline函数不保存换行符,但是我们有些时候需要保留换行符,所以我们提供这样一个选择
            while (std::getline(in, line))
            {
                (*content) += line;
                (*content) += (retain_LF ? "\n" : "");
            }

            in.close();
            return true;
        }
    };

    class string_util
    {
    public:
        /*
         *src:输入型参数:需要进行切分的字符串
         * target:输出型参数:保存切分好的字符串
         * separator:指定分隔符
         */

        static void split_string(const std::string &src, std::vector<std::string> *target, const std::string &separator)
        {
            boost::split((*target), src, boost::is_any_of(separator), boost::algorithm::token_compress_on);
        }
    };
}
